tensor([[[[[0.0800, 0.0700],
           [0.0700, 0.0800]],

          [[0.0400, 0.0200],
           [0.0200, 0.0400]]],


         [[[0.0400, 0.0200],
           [0.0200, 0.0400]],

          [[0.7800, 0.8600],
           [0.8900, 0.8600]]]],



        [[[[0.3600, 0.3600],
           [0.3600, 0.3600]],

          [[0.3700, 0.3800],
           [0.3800, 0.3700]]],


         [[[0.3700, 0.3800],
           [0.3800, 0.3700]],

          [[0.0000, 0.1000],
           [0.0000, 0.0400]]]],



        [[[[0.2300, 0.2300],
           [0.2300, 0.2300]],

          [[0.2400, 0.2400],
           [0.2400, 0.2400]]],


         [[[0.2400, 0.2400],
           [0.2400, 0.2400]],

          [[0.1500, 0.0300],
           [0.0000, 0.0600]]]],



        [[[[0.3300, 0.3400],
           [0.3400, 0.3300]],

          [[0.3500, 0.3500],
           [0.3500, 0.3500]]],


         [[[0.3500, 0.3500],
           [0.3500, 0.3500]],

          [[0.0700, 0.0100],
           [0.1100, 0.0400]]]]], grad_fn=<RoundBackward1>)